package logic

import (
	"context"
	"time"

	"probe-users/internal/errorx"
	"probe-users/internal/svc"
	"probe-users/internal/utils"
	"probe-users/pb/users"

	"github.com/pkg/errors"
	"github.com/zeromicro/go-zero/core/logx"
)

type UpdateUserLogic struct {
	ctx    context.Context
	svcCtx *svc.ServiceContext
	logx.Logger
}

func NewUpdateUserLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UpdateUserLogic {
	return &UpdateUserLogic{
		ctx:    ctx,
		svcCtx: svcCtx,
		Logger: logx.WithContext(ctx),
	}
}

func (l *UpdateUserLogic) UpdateUser(in *users.UpdateUserRequest) (*users.UpdateUserResponse, error) {
	// 参数验证
	if err := l.validateUpdateUserParams(in); err != nil {
		return nil, err
	}

	// 解析Token
	claims, err := utils.ParseToken(in.Token, l.svcCtx.Config.JWTAuth.AccessSecret)
	if err != nil {
		return nil, err
	}

	// 确定要更新的用户ID
	var targetUserId int64
	if in.UserId == 0 {
		// 如果没有指定用户ID，则更新当前用户
		targetUserId = claims.UserId
	} else {
		// 检查是否有权限更新其他用户
		hasPermission, err := l.svcCtx.Enforcer.Enforce(claims.RoleId, "user", "update:other")
		if err != nil {
			return nil, errors.Wrapf(err, "检查权限失败")
		}
		if !hasPermission {
			return nil, errorx.ErrPermissionDenied
		}
		targetUserId = in.UserId
	}

	// 查询用户
	user, err := l.svcCtx.UserModel.FindOne(targetUserId)
	if err != nil {
		return nil, errors.Wrapf(err, "查询用户失败: %d", targetUserId)
	}

	// 检查邮箱是否已被其他用户使用
	if in.Email != "" && in.Email != user.Email {
		existingUser, err := l.svcCtx.UserModel.FindByEmail(in.Email)
		if err != nil {
			// 如果是未找到用户的错误，则说明邮箱未被占用，可以继续
			if errors.Is(err, errorx.ErrUserNotFound) {
				user.Email = in.Email
			} else {
				// 其他错误需要返回
				return nil, errors.Wrapf(err, "查询邮箱失败: %s", in.Email)
			}
		} else {
			// 找到了用户，检查是否是同一个用户
			if existingUser.ID != user.ID {
				return nil, errorx.ErrEmailExists
			}
			user.Email = in.Email
		}
	}

	// 检查手机号是否已被其他用户使用
	if in.Phone != "" && in.Phone != user.Phone {
		existingUser, err := l.svcCtx.UserModel.FindByPhone(in.Phone)
		if err != nil {
			// 如果是未找到用户的错误，则说明手机号未被占用，可以继续
			if errors.Is(err, errorx.ErrUserNotFound) {
				user.Phone = in.Phone
			} else {
				// 其他错误需要返回
				return nil, errors.Wrapf(err, "查询手机号失败: %s", in.Phone)
			}
		} else {
			// 找到了用户，检查是否是同一个用户
			if existingUser.ID != user.ID {
				return nil, errorx.ErrPhoneExists
			}
			user.Phone = in.Phone
		}
	}

	// 更新昵称
	if in.Nickname != "" {
		user.Nickname = in.Nickname
	}

	// 更新时间
	user.UpdatedAt = time.Now()

	// 保存更新
	if err := l.svcCtx.UserModel.Update(user); err != nil {
		return nil, errors.Wrapf(err, "更新用户失败: %d", user.ID)
	}

	// 获取角色名称
	role, err := l.svcCtx.RoleModel.FindOne(user.RoleID)
	if err != nil {
		return nil, errors.Wrapf(err, "查询角色失败: %d", user.RoleID)
	}

	return &users.UpdateUserResponse{
		Success: true,
		User: &users.User{
			Id:        user.ID,
			Username:  user.Username,
			Email:     user.Email,
			Phone:     user.Phone,
			Nickname:  user.Nickname,
			RoleId:    user.RoleID,
			RoleName:  role.Name,
			CreatedAt: user.CreatedAt.Unix(),
			UpdatedAt: user.UpdatedAt.Unix(),
		},
	}, nil
}

func (l *UpdateUserLogic) validateUpdateUserParams(in *users.UpdateUserRequest) error {
	if in.Token == "" {
		return errorx.ErrTokenNotEmpty
	}
	return nil
}
